import matplotlib.pyplot as plt
import numpy as np


if __name__ == '__main__':
    x = np.arange(1, 11)
    y = np.random.randint(1000, 5000, size=len(x))
    y2 = np.random.randint(500, 3000, size=len(x))

    # 绘图时添加标签
    plt.plot(x, y, linewidth=1, label='收入')
    plt.plot(x, y2, linewidth=1, label='支出')
    plt.legend()

    # 显示具体数据
    for a, b in zip(x, y):
        # 在坐标(a, b)处添加string信息b
        plt.text(a, b, f'{b}元')
    for a, b in zip(x, y2):
        plt.text(a, b, f'{b}元')

    # 修改默认字体以显示中文
    plt.rcParams['font.sans-serif'] = ['SimHei']
    # 取消使用unicode的负号以显示负号
    plt.rcParams['axes.unicode_minus'] = False

    # 设置xy轴名称和字体大小
    plt.xlabel('x 轴', fontsize=12)
    plt.ylabel('y 轴', fontsize=12)
    plt.title('y是x的平方')
    plt.show()
